iT邦幫忙

2024 iThome 鐵人賽

DAY 2
0
AI/ ML & Data

AI 學習紀錄系列 第 2

Day 2: 多模態模型測試,架在 colab 上

  • 分享至 

  • xImage
  •  

今天想嘗試架看看多模態的模型,一樣找一個中文的來試試看。
https://github.com/THUDM/VisualGLM-6B

原先想架在自家電腦上,但發現架在windows上比較麻煩,有些套件不好裝,決定改到 colab 上運行。

到 colab 上開一個 ipynb,調整執行階段類型,選有 GPU 來用,因為我自己有買 colab pro,所以再多勾了一個大量 RAM,RAM 不夠的做法,改天再研究。

https://ithelp.ithome.com.tw/upload/images/20240819/20168318YlAP25JHke.png
https://ithelp.ithome.com.tw/upload/images/20240819/20168318Gd9has8B93.png

git 中有提供 requirements.txt ,直接照著裝會有些套件版本衝突問題,我調整了其中兩項,SwissArmyTransformer==0.4.5

transformers==4.33.2

這是最後的 requirements.txt 的模樣,紀錄一下。

SwissArmyTransformer==0.4.5
torch>1.10.0
torchvision
transformers==4.33.2
mdtex2html
gradio

可以把 requirements.txt 直接丟上檔案區,直接下指令裝。
https://ithelp.ithome.com.tw/upload/images/20240819/20168318q56OK5A8Nv.png
https://ithelp.ithome.com.tw/upload/images/20240819/20168318yl0znaYxOD.png

裝完發現還是有版本衝突,先把 cudf-cu12 降板。
https://ithelp.ithome.com.tw/upload/images/20240819/20168318Ew4GvduixE.png

降板後會把 pyarrow 、protobuf 一併更新,這裡 pyarrow 被降到 14.0.2,符合 ibis-framework 8.0.0 的要求了,但 datasets 、tensorflow-metadata 有衝突,先處理 datasets。
https://ithelp.ithome.com.tw/upload/images/20240819/201683183jQJ3RIJHH.png
https://ithelp.ithome.com.tw/upload/images/20240819/201683188SR8E1H4go.png

把 datasets 降板,發現他把 fsspec 也降板了,使的 gcsfs 得降板,那就把 gcsfs 也降,順利解決。
https://ithelp.ithome.com.tw/upload/images/20240819/20168318DMqCQuCFfJ.png
https://ithelp.ithome.com.tw/upload/images/20240819/20168318ZocpbqLhef.png
https://ithelp.ithome.com.tw/upload/images/20240819/20168318FauSlRg9fM.png
https://ithelp.ithome.com.tw/upload/images/20240819/201683185edEgdJCls.png

最後是還沒處理的 tensorflow-metadata,繼續降板,這樣就差不多了。
https://ithelp.ithome.com.tw/upload/images/20240819/20168318H2iIKBnnSF.png

這邊先停一下,到 huggingface 註冊帳號,申請 Access Tokens。
https://huggingface.co/settings/tokens

輸入好名稱,權限可以先都不開,拉到底點 Create token。
https://ithelp.ithome.com.tw/upload/images/20240819/20168318kwxBlpzegf.png
https://ithelp.ithome.com.tw/upload/images/20240819/20168318wk1kZoyL9Q.png
https://ithelp.ithome.com.tw/upload/images/20240819/20168318NMLFM1njbd.png

記下 token,回到 colab,在左手邊密鑰區,新增一個 HF_TOKEN,並把剛剛的 token 貼到值,並開給他筆記本存取權。這在稍後 transformers 去使用 huggingface 的 model 會用到。
https://ithelp.ithome.com.tw/upload/images/20240819/20168318gAW5LTN7bK.png

如果沒申請,或是沒給筆記本存取權,也會有提醒,很貼心。
https://ithelp.ithome.com.tw/upload/images/20240819/201683189JrpkLAqow.png
https://ithelp.ithome.com.tw/upload/images/20240819/20168318iYvKD22hyK.png

接著就可以使用 README 的範例來測試了,可以上傳一張影像到 colab 中,your image path 改成自己的影像名
https://ithelp.ithome.com.tw/upload/images/20240819/2016831879dRSFKU4X.png
https://ithelp.ithome.com.tw/upload/images/20240819/20168318hBqn9X1MQp.png

會得到這樣的結果:
https://ithelp.ithome.com.tw/upload/images/20240819/20168318B1PRMNaVIh.png
https://ithelp.ithome.com.tw/upload/images/20240819/20168318SACCy0hRgg.png


上一篇
Day 1: 環境
下一篇
Day 3: Gemini API
系列文
AI 學習紀錄30
圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

尚未有邦友留言

立即登入留言